import plotly.express as px
import pandas as pd

import csv

#探索数据的结构
filename='D:\GITEE\study_code\world_fires_1_day.csv'
with open(filename) as f:
    reader=csv.reader(f)
    header_row=next(reader)
    #for index,column_header in enumerate(header_row):
        #print(index,column_header)
    #根据上段代码，得出我们所需索引为0,1,2
    #从文件中获取信息，得到所需经度,纬度和火灾强度
    lat,lon,bri=[],[],[]
    for row in reader:
        lat.append(float(row[0]))
        lon.append(float(row[1]))
        bri.append(float(row[2]))

#创建DataFrame封装数据
data=pd.DataFrame(
    data=zip(lat,lon,bri),columns=['经度','纬度','火灾强度']
)
data.head()

#根据已有数据和需求绘制图形
fig=px.scatter(
    data,
    x='经度',
    y='纬度',
    range_x=[-200,200],
    range_y=[-90,90],
    width=800,
    height=800,
    title='全球火灾散点图',
    size='火灾强度',
    size_max=10,
    color='火灾强度',
)
fig.write_html('global_fire.html')

#绘制图形
fig.show()